from .mutr import build
from .tlggn.tlggnv1 import build_tlggnv1
from .tlggn.tlggnv2 import build_tlggnv2
from .tlggn.tlggnv3 import build_tlggnv3
def build_model(args):
    if args.me == 'tlggnv1':
        return build_tlggnv1(args)
    elif args.me == 'tlggnv2':
        return build_tlggnv2(args)
    elif args.me == 'tlggnv3':
        return build_tlggnv3(args)
    return build(args)
